package init

import (
	"flag"
	"getaway/basic/config"
	ai "getaway/basic/proto/aiproto"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"log"
)

func AiGrpc() {
	flag.Parse()
	// 设置到行为追踪服务的连接
	conn, err := grpc.NewClient("127.0.0.1:50059", grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		log.Printf("连接行为追踪服务失败: %v", err)
		// 注意：这里不中断程序，因为行为追踪服务不可用时应该降级使用日志
		return
	}

	// 初始化客户端
	config.AiClient = ai.NewAiClient(conn)
	log.Println("AI服务客户端初始化成功")

	// 这里不关闭连接，因为需要长期保持连接以进行服务调用
}
